/*
 * 
 * 
 * 
 */
package com.cms.template.directive;

import com.cms.entity.Site;
import com.cms.service.SiteService;
import freemarker.core.Environment;
import freemarker.template.TemplateDirectiveBody;
import freemarker.template.TemplateException;
import freemarker.template.TemplateModel;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.util.List;
import java.util.Map;

/**
 * 模板指令 - 站点列表
 * 
 * 
 * 
 */
@Component
public class SiteListDirective extends BaseDirective {

	@Autowired
	private SiteService siteService;

	/** 变量名称 */
	private static final String VARIABLE_NAME = "sites";

	@SuppressWarnings({ "unchecked", "rawtypes" })
	public void execute(Environment env, Map params, TemplateModel[] loopVars, TemplateDirectiveBody body) throws TemplateException, IOException {
		List<Site> sites = siteService.findAll();
		setLocalVariable(VARIABLE_NAME, sites, env, body);
	}
}